import pytest
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.firefox.service import Service

from utils import DriverUtil, get_alert_msg


class TestLogin(object):
    def setup_class(self):
        self.driver = DriverUtil.get_driver()  # 获取浏览器对象

    def teardown_class(self):
        DriverUtil.quit_driver()  # 退出浏览器对象

    def test_account_does_not_exist(self):
        # 1、点击首页的"登录"链接，进入登录页面
        self.driver.find_element(By.LINK_TEXT, "登录").click()
        # 2、输入一个不存在的用户名
        self.driver.find_element(By.ID, "username").send_keys("13811110001")
        # 3、输入密码
        self.driver.find_element(By.ID, "password").send_keys("123456")
        # 4、输入验证码
        self.driver.find_element(By.ID, 'verify_code').send_keys("8888")
        # 5、点击登录按钮
        self.driver.find_element(By.NAME, "sbtbutton").click()
        # 6、获取错误提示信息
        msg = get_alert_msg()
        print("错误信息为：", msg)

    def test_wrong_password(self):
        # 1、点击首页的"登录"链接，进入登录页面
        self.driver.find_element(By.LINK_TEXT, '登录').click()
        # 2、输入用户名
        self.driver.find_element(By.ID, 'username').send_keys("13800001111")
        # 3、输入一个错误的密码
        self.driver.find_element(By.ID, 'password').send_keys("123456")
        # 4、输入验证码
        self.driver.find_element(By.ID, 'verify_code').send_keys("8888")
        # 5、点击登录按钮
        self.driver.find_element(By.NAME, 'sbtbutton').click()
        # 6、获取错误提示信息
        msg = get_alert_msg()
        print("错误信息为：", msg)

if __name__ == '__main__':
    pytest.main(['-s', 'tpshop_login3'])